package com.yzc.aliOSS;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.VoidResult;
import com.yzc.model.entity.AliOSS.AliOSSProperties;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

/**
 * 阿里云 OSS 工具类
 */
@Component       //文件上传方法
public class AliOSSUtils {

    //注入配置参数实体类对象
    @Autowired
    private AliOSSProperties aliOSSProperties;

    /**
     * 实现单个文件上传到OSS
     */
    public String uploadFile(MultipartFile file) throws IOException {
        // 获取上传的文件的输入流
        InputStream inputStream = file.getInputStream();

        // 避免文件覆盖
        String originalFilename = file.getOriginalFilename();
        String fileName = UUID.randomUUID().toString() + originalFilename.substring(originalFilename.lastIndexOf("."));

        //上传文件到 OSS
        OSS ossClient = new OSSClientBuilder().build(aliOSSProperties.getEndpoint(),
                aliOSSProperties.getAccessKeyId(), aliOSSProperties.getAccessKeySecret());
        ossClient.putObject(aliOSSProperties.getBucketName(), fileName, inputStream);

        //文件访问路径
        String url =aliOSSProperties.getEndpoint().split("//")[0] + "//" + aliOSSProperties.getBucketName() + "." + aliOSSProperties.getEndpoint().split("//")[1] + "/" + fileName;

        // 关闭ossClient
        ossClient.shutdown();
        return url;// 把上传到oss的路径返回
    }
    /**
     * 多个文件上传到OSS
     */
     public String uploadFiles(List<MultipartFile> files) throws IOException {
        List<String> urls = new ArrayList<>();
        // 创建OSSClient实例
        OSS ossClient = new OSSClientBuilder().build(
                aliOSSProperties.getEndpoint(),
                aliOSSProperties.getAccessKeyId(),
                aliOSSProperties.getAccessKeySecret()
        );
        try {
            for (MultipartFile file : files) {
                // 获取上传的文件的输入流
                InputStream inputStream = file.getInputStream();

                // 避免文件覆盖
                String originalFilename = file.getOriginalFilename();
                String fileName = UUID.randomUUID().toString() + originalFilename.substring(originalFilename.lastIndexOf("."));

                // 上传文件到OSS
                ossClient.putObject(aliOSSProperties.getBucketName(), fileName, inputStream);

                // 文件访问路径
                String url = aliOSSProperties.getEndpoint().split("//")[0] + "//" +
                        aliOSSProperties.getBucketName() + "." +
                        aliOSSProperties.getEndpoint().split("//")[1] + "/" + fileName;

                // 添加URL到列表
                urls.add(url);
            }
        } finally {
            // 关闭ossClient
            ossClient.shutdown();
        }

        return String.join(",", urls);
    }
    /**
     * 删除OSS中的文件
     * @param objectName 文件名
     */
    public Boolean deleteFile(String objectName) {
        // 创建OSSClient实例
        OSS ossClient = new OSSClientBuilder().build(
                aliOSSProperties.getEndpoint(),
                aliOSSProperties.getAccessKeyId(),
                aliOSSProperties.getAccessKeySecret()
        );
        try {
            // 删除文件
            ossClient.deleteObject(aliOSSProperties.getBucketName(), objectName);
            return true;
        } catch (Exception e){
            e.printStackTrace();
            return false;
        }
        finally {
            // 关闭ossClient
            ossClient.shutdown();
        }
    }
}
